#!/bin/bash

# 水文预报 WebSocket 服务器启动脚本（本地版本）
# 作者：Gaoyu <gaoyu.ssdut@icloud.com>

set -e

echo "🚀 启动水文预报 WebSocket 服务器（本地版本）..."
echo "=============================================="

# 检查 .env 文件
if [ ! -f ".env" ]; then
    echo "⚠️  .env 文件不存在，从示例文件创建..."
    if [ -f ".env.example" ]; then
        cp .env.example .env
        echo "✅ 已从 .env.example 创建 .env 文件"
        echo "📝 注意：VLLM_BASE_URL 已设置为 http://10.48.0.81:8000/v1"
    else
        echo "❌ 错误：.env.example 文件不存在"
        exit 1
    fi
fi

# 显示配置
echo ""
echo "📋 当前配置:"
grep -v "^#" .env | grep -v "^$" || true
echo ""
echo "=============================================="

# 检查 Python 环境
if ! command -v python &> /dev/null; then
    echo "❌ 错误：未找到 Python"
    exit 1
fi

echo "🐍 Python 版本: $(python --version)"

# 检查依赖
if ! python -c "import websockets" 2>/dev/null; then
    echo "⚠️  警告：未安装 websockets，尝试安装依赖..."
    pip install -r requirements.txt
fi

echo ""
echo "🚀 启动服务..."
echo "=============================================="

# 启动服务
python src/websocket_server.py


